# encoding=utf-8
# 示例参考: /home/HwHiAiUser/samples/notebooks/04-image-HDR-enhance

"""
atc --model=./image_HDR_enhance.pb
    --framework=3
    --output=image_HDR_enhance
    --soc_version=Ascend310B1
    --input_shape="input:1,512,512,3"
    --input_format=NHWC
    --output_type=FP32

其中各个参数具体含义如下：
    --model：原始模型文件。
    --framework：原始框架类型，0:Caffe; 1:MindSpore; 2:Tensorflow; 5:Onnx。
    --output：离线推理om模型文件路径。
    --soc_version：昇腾AI处理器型号，填写"Ascend310B1"。
    --input_shape：模型输入节点名称和shape。
    --input_format：输入Tensor的内存排列方式。
    --output_type：指定网络输出数据类型或指定某个输出节点的输出类型。
"""
import os

command_line = ("atc --model=./export_funds_stillgan.onnx \
                    --framework=5 \
                    --output=funds_om_enhance \
                    --soc_version=Ascend310B1 \
                    --input_shape='input:1,3,512,512' \
                    --input_format=NCHW \
                    --output_type=FP32 ")

print(command_line)
# 大约10分钟
print(os.system(command_line))